# 105.从前序与中序遍历序列构造二叉树
# 给定两个整数数组preorder和inorder ，其中preorder是二叉树的先序遍历， inorder是同一棵树的中序遍历，请构造二叉树并返回其根节点。
#
# 示例1:
# 输入: preorder = [3, 9, 20, 15, 7], inorder = [9, 3, 15, 20, 7]
# 输出: [3, 9, 20, null, null, 15, 7]
#
# 示例2:
# 输入: preorder = [-1], inorder = [-1]
# 输出: [-1]


class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
class Solution:
    def buildTree(self, preorder: [int], inorder: [int]) -> [TreeNode]:
        # 第一步：如果数组大小为零的话，说明是空节点了。
        def traver(preorder,inorder):
            if not preorder:
                return None
            # 第二步：如果不为空，那么取前序数组第一个元素作为节点元素。
            root_val = preorder[0]
            root = TreeNode(root_val)
            # 叶子节点要处理
            if len(preorder) == 1:
                return root
            # 第三步：找到前序数组第一个元素在中序数组的位置，作为切割点
            d_index = inorder.index(root_val)
            # 第四步：切割中序数组，切成中序左数组和中序右数组 （顺序别搞反了，一定是先切中序数组）
            left_inorder = inorder[:d_index]
            right_inorder = inorder[d_index+1:]
            # 第五步：切割前序数组，切成前序左数组和前序右数组
            preorder = preorder[1:]
            left_preorder = preorder[:len(left_inorder)]
            right_preorder = preorder[len(left_inorder):]
            # 第六步：递归处理左区间和右区间
            root.left = traver(left_preorder,left_inorder)
            root.right = traver(right_preorder,right_inorder)
            return root
        if not preorder or not inorder:
            return None
        return traver(preorder,inorder)


if __name__ == '__main__':
    preorder = [3,9,20,15,7]
    inorder = [9,3,15,20,7]
    tmp = Solution()
    res = tmp.buildTree(preorder,inorder)
    print(res)